from aiClients.baseAiClient import BaseAIClient


class ModelContext:
    def __init__(self):
        self.model = None

    def set_model(self, model: BaseAIClient):
        self.model = model

    def query(self, prompt: str) -> str:
        if not self.model:
            raise ValueError("No model is set")
        return self.model.query(prompt)